import torch
print(torch.cuda.device_count())
ckpt = torch.load("251025_cifar100_RANet_BS64_anytime_CDM_cuda_0_LR0.1/save_models/model_best.pth.tar")
print(ckpt)
